numpy
torch
matplotlib
ipython
ipykernel
